import matplotlib.pyplot as plt
import imageio
import random
#plt.hold(False)

file = open("submission.csv", "r").read().splitlines()[1:]

start = 30
stop = min(start+10, len(file)-1)
lines = file[start:stop]

for line in lines:
    l, r = line.split(',')
    plt.figure(figsize=(3,3))
    query_path = 'test'+'/'+l+'.jpg'
    images = ['index'+'/'+i+'.jpg' for i in r.split()]
    # Show the query image
    plt.imshow(imageio.imread(query_path))
    plt.figure(figsize=(20,10))
    
    # Then show 4 of the similar images
    columns = 5
    for i, image in enumerate(images):
        plt.subplot(len(images) / columns + 1, columns, i + 1)
        plt.imshow(imageio.imread(image))